# Add all the constants here


# WM type
WM_KGW = 'KGW'
WM_UNIGRAM = 'unigram'
WM_AAR = 'AAR'
WM_EXP = 'EXP'

# Model name
LLAMA7B = 'meta-llama/Llama-2-7b-hf'
LLAMA70B = "meta-llama/Meta-Llama-3-70B"
QWEN72B = "Qwen/Qwen2-72B"
GEMMA2B = "google/gemma-2-2b-it"
GEMMA9B = 'google/gemma-2-9b-it'
QWEN7B = "Qwen/Qwen2-7B-Instruct"

# dataset name
DATA_REALNEWS = 'realnews'
DATA_EN = 'en' 
DATA_LFQA = 'lfqa'


# dataset size
REALNEWS_SIZE=100 # only use 100 samples for testing
LFQA_SIZE=100 # only use 100 samples for testing
EN_SIZE = 200000
EXTRA_EN_SIZE = 50000

#EN_SIZE = 100 # test mode: temporarily use 100 samples for testing


# text length
PROMPT_LEN=50
NEW_TOKEN_LEN=200
MAX_NEW_TOKENS=NEW_TOKEN_LEN
MIN_NEW_TOKENS=NEW_TOKEN_LEN-50
PROCESS_NUM=20